#!/usr/bin/env bash

# Train and test STAViS audiovisual models for all splits

base_path='./experiments'

# split=1
# python main_one_dataset.py --gpu_devices 0, --batch_size 32 --n_threads 12 \
#     --dataset "coutrot2" \
#     --no_train \
#     --no_test \
#     --pool_layer "tempAvg" \
#     --learning_rate 0.01 \
#     --temp_reduction 2 \
#     --use_transformer \
#     --num_heads 8 \
#     --attn_dropout 0.1 \
#     --lr_transformer 0.001 \
#     --upsample_func "bilinear" \
#     --no_use_transposeConv \
#     --no_use_spatio_att \
#     --no_use_dsam_att \
#     --no_use_dsam_multiScale \
#     --audiovisual True \
#     --root_path ${base_path} \
#     --resume_path ./experiments/202011300004/save_best.pth \
#     --annotation_path_coutrot2_train ./data/fold_lists/Coutrot_db2_list_train_${split}_fps.txt \
#   	--annotation_path_coutrot2_test ./data/fold_lists/Coutrot_db2_list_test_${split}_fps.txt

# split=2
# python main_one_dataset.py --gpu_devices 0, --batch_size 32 --n_threads 128 \
#     --dataset "coutrot2" \
#     --no_train \
#     --no_test \
#     --pool_layer "tempAvg" \
#     --learning_rate 0.01 \
#     --temp_reduction 2 \
#     --use_transformer \
#     --num_heads 8 \
#     --attn_dropout 0.1 \
#     --lr_transformer 0.001 \
#     --upsample_func "bilinear" \
#     --no_use_transposeConv \
#     --no_use_spatio_att \
#     --no_use_dsam_att \
#     --no_use_dsam_multiScale \
#     --audiovisual True \
#     --root_path ${base_path} \
#     --resume_path ./experiments/202011300004/save_best.pth \
#     --annotation_path_coutrot2_train ./data/fold_lists/Coutrot_db2_list_train_${split}_fps.txt \
#   	--annotation_path_coutrot2_test ./data/fold_lists/Coutrot_db2_list_test_${split}_fps.txt


# split=3
# python main_one_dataset.py --gpu_devices 0, --batch_size 32 --n_threads 12 \
#     --dataset "coutrot2" \
#     --no_train \
#     --no_test \
#     --pool_layer "tempAvg" \
#     --learning_rate 0.01 \
#     --temp_reduction 2 \
#     --use_transformer \
#     --num_heads 8 \
#     --attn_dropout 0.1 \
#     --lr_transformer 0.001 \
#     --upsample_func "bilinear" \
#     --no_use_transposeConv \
#     --no_use_spatio_att \
#     --no_use_dsam_att \
#     --no_use_dsam_multiScale \
#     --audiovisual True \
#     --root_path ${base_path} \
#     --resume_path ./experiments/202011300004/save_best.pth \
#     --annotation_path_coutrot2_train ./data/fold_lists/Coutrot_db2_list_train_${split}_fps.txt \
#   	--annotation_path_coutrot2_test ./data/fold_lists/Coutrot_db2_list_test_${split}_fps.txt

# # wo audio
# split=1
# python main_one_dataset.py --gpu_devices 0, --batch_size 32 --n_threads 12 \
#     --dataset "coutrot2" \
#     --no_train \
#     --no_test \
#     --pool_layer "tempAvg" \
#     --learning_rate 0.01 \
#     --temp_reduction 2 \
#     --upsample_func "bilinear" \
#     --no_use_transposeConv \
#     --no_use_spatio_att \
#     --audiovisual "" \
#     --root_path ${base_path} \
#     --pretrain_path ./experiments/202011200247/save_40.pth \
#     --annotation_path_coutrot2_train ./data/fold_lists/Coutrot_db2_list_train_${split}_fps.txt \
#   	--annotation_path_coutrot2_test ./data/fold_lists/Coutrot_db2_list_test_${split}_fps.txt

# split=2
# python main_one_dataset.py --gpu_devices 0, --batch_size 32 --n_threads 128 \
#     --dataset "coutrot2" \
#     --no_train \
#     --no_test \
#     --pool_layer "tempAvg" \
#     --learning_rate 0.01 \
#     --temp_reduction 2 \
#     --upsample_func "bilinear" \
#     --no_use_transposeConv \
#     --no_use_spatio_att \
#     --audiovisual "" \
#     --root_path ${base_path} \
#     --pretrain_path ./experiments/202011200335/save_40.pth \
#     --annotation_path_coutrot2_train ./data/fold_lists/Coutrot_db2_list_train_${split}_fps.txt \
#   	--annotation_path_coutrot2_test ./data/fold_lists/Coutrot_db2_list_test_${split}_fps.txt


# split=3
# python main_one_dataset.py --gpu_devices 0, --batch_size 32 --n_threads 12 \
#     --dataset "coutrot2" \
#     --no_train \
#     --no_test \
#     --pool_layer "tempAvg" \
#     --learning_rate 0.01 \
#     --temp_reduction 2 \
#     --upsample_func "bilinear" \
#     --no_use_transposeConv \
#     --no_use_spatio_att \
#     --audiovisual "" \
#     --root_path ${base_path} \
#     --pretrain_path ./experiments/202011200421/save_40.pth \
#     --annotation_path_coutrot2_train ./data/fold_lists/Coutrot_db2_list_train_${split}_fps.txt \
#   	--annotation_path_coutrot2_test ./data/fold_lists/Coutrot_db2_list_test_${split}_fps.txt

split=1
python main_oneWhole_dataset.py --gpu_devices 0, --batch_size 32 --n_threads 12 \
    --dataset "coutrot2" \
    --no_train \
    --no_test \
    --pool_layer "tempAvg" \
    --learning_rate 0.01 \
    --temp_reduction 2 \
    --upsample_func "bilinear" \
    --no_use_transposeConv \
    --no_use_spatio_att \
    --audiovisual "" \
    --root_path ${base_path} \
    --pretrain_path ./experiments/202011200247/save_40.pth \
    --pretrain_path_split1 ./experiments/202011200247/save_40.pth \
    --pretrain_path_split2 ./experiments/202011200335/save_40.pth \
    --pretrain_path_split3 ./experiments/202011200421/save_40.pth \
    --annotation_path_coutrot1_train ./data/fold_lists/Coutrot_db1_list_train_${split}_fps.txt \
    --annotation_path_coutrot1_test ./data/fold_lists/Coutrot_db1_list_test_${split}_fps.txt \
    --annotation_path_coutrot2_train ./data/fold_lists/Coutrot_db2_list_train_${split}_fps.txt \
    --annotation_path_coutrot2_test ./data/fold_lists/Coutrot_db2_list_test_${split}_fps.txt \
    --annotation_path_summe_train ./data/fold_lists/SumMe_list_train_${split}_fps.txt \
    --annotation_path_summe_test ./data/fold_lists/SumMe_list_test_${split}_fps.txt \
    --annotation_path_etmd_train ./data/fold_lists/ETMD_av_list_train_${split}_fps.txt \
    --annotation_path_etmd_test ./data/fold_lists/ETMD_av_list_test_${split}_fps.txt \
    --annotation_path_avad_train ./data/fold_lists/AVAD_list_train_${split}_fps.txt \
    --annotation_path_avad_test ./data/fold_lists/AVAD_list_test_${split}_fps.txt
